package com.promote.algorithm;

/**
 * @ClassName MergeTwoLists
 * Description 合并两个有序链表
 * @Author LiZiHao
 * Date 2022/3/30 10:59
 * @Version 1.0
 **/
public class MergeTwoLists {

    public class ListNode {
        int val;
        ListNode next;

        ListNode() {
        }

        ListNode(int val) {
            this.val = val;
        }

        ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }


    /**
     * 因为链表是升序的，我们只需要遍历每个链表的头，比较一下哪个小就把哪个链表的头拿出来放到新的链表中，一直这样循环，直到有一个链表为空，然后我们再把另一个不为空的链表挂到新的链表中。
     *
     * @param list1
     * @param list2
     * @return
     */
    public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        //空判断
        if (list1 == null) {
            return list2;
        }
        if (list2 == null) {
            return list1;
        }

        ListNode dummy = new ListNode(0);
        ListNode curr = dummy;

        while (list1 != null && list2 != null) {
            //对链表进行比较
            if (list1.val <= list2.val) {
                curr.next = list1;
                list1 = list1.next;
            } else {
                curr.next = list2;
                list2 = list2.next;
            }
            curr = curr.next;
        }
        //把不为空的链表挂到新的链表上
        curr.next = list1 == null ? list2 : list1;
        return dummy.next;
    }

    /**
     * 递归的方式
     *
     * @param list1
     * @param list2
     * @return
     */
    public ListNode mergeTwoLists1(ListNode list1, ListNode list2) {
        //只要有一个为空，就返回另一个
        if (list1 == null || list2 == null) {
            return list2 == null ? list1 : list2;
        }

        //把小的赋值新的链表
        ListNode res = (list2.val < list1.val) ? list2 : list1;
        res.next = mergeTwoLists1(res.next, res == list1 ? list2 : list1);
        return res;
    }
}
